home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Extensions / Imaging / PIL / TgaImagePlugin.py < prev    next >
Encoding:
Text File  |  2000-06-23  |  2.8 KB  |  134 lines
  1. #
  2. # The Python Imaging Library.
  3. # $Id: TgaImagePlugin.py,v 1.1.1.2 1999/01/13 09:39:56 sjoerd Exp $
  4. #
  5. # TGA file handling
  6. #
  7. # History:
  8. # 95-09-01 fl    created (reads 24-bit files only)
  9. # 97-01-04 fl    support more TGA versions, including compressed images
  10. # 98-07-04 fl    fixed orientation and alpha layer bugs
  11. # 98-09-11 fl    fixed orientation for runlength decoder
  12. #
  13. # Copyright (c) Secret Labs AB 1997-98.
  14. # Copyright (c) Fredrik Lundh 1995-97.
  15. #
  16. # See the README file for information on usage and redistribution.
  17. #
  18.  
  19.  
  20. __version__ = "0.3"
  21.  
  22. import Image, ImageFile, ImagePalette
  23.  
  24.  
  25. def i16(c):
  26.     return ord(c[0]) + (ord(c[1])<<8)
  27.  
  28. def i32(c):
  29.     return ord(c[0]) + (ord(c[1])<<8) + (ord(c[2])<<16) + (ord(c[3])<<24)
  30.  
  31.  
  32. MODES = {
  33.     # map imagetype/depth to rawmode
  34.     (1, 8):  "P",
  35.     (3, 1):  "1",
  36.     (3, 8):  "L",
  37.     (2, 16): "BGR;5",
  38.     (2, 24): "BGR",
  39.     (2, 32): "BGRA",
  40. }
  41.  
  42.  
  43. def _accept(prefix):
  44.     return prefix[0] == "\0"
  45.  
  46.  
  47. class TgaImageFile(ImageFile.ImageFile):
  48.  
  49.     format = "TGA"
  50.     format_description = "Targa"
  51.  
  52.     def _open(self):
  53.  
  54.  
  55.  
  56.     # process header
  57.     s = self.fp.read(18)
  58.  
  59.     id = ord(s[0])
  60.  
  61.     colormaptype = ord(s[1])
  62.     imagetype = ord(s[2])
  63.  
  64.     depth = ord(s[16])
  65.  
  66.     flags = ord(s[17])
  67.  
  68.     self.size = i16(s[12:]), i16(s[14:])
  69.  
  70.     # validate header fields
  71.     if id != 0 or colormaptype not in (0, 1) or\
  72.        self.size[0] <= 0 or self.size[1] <= 0 or\
  73.        depth not in (8, 16, 24, 32):
  74.         raise SyntaxError, "not a TGA file"
  75.  
  76.     # image mode
  77.     if imagetype in (3, 11):
  78.         self.mode = "L"
  79.         if depth == 1:
  80.         self.mode = "1" # ???
  81.     elif imagetype in (1, 9):
  82.         self.mode = "P"
  83.     elif imagetype in (2, 10):
  84.         self.mode = "RGB"
  85.         if depth == 32:
  86.         self.mode = "RGBA"
  87.         else:
  88.         raise SyntaxError, "unknown TGA mode"
  89.  
  90.     # orientation
  91.     orientation = flags & 0x30
  92.     if orientation == 0x20:
  93.         orientation = 1
  94.     elif not orientation:
  95.         orientation = -1
  96.     else:
  97.         raise SyntaxError, "unknown TGA orientation"
  98.  
  99.     if imagetype & 8:
  100.         self.info["compression"] = "tga_rle"
  101.  
  102.     if colormaptype:
  103.         # read palette
  104.         start, size, mapdepth = i16(s[3:]), i16(s[5:]), i16(s[7:])
  105.         if mapdepth == 16:
  106.         self.palette = ImagePalette.raw("BGR;16",
  107.             "\0"*2*start + self.fp.read(2*size))
  108.         elif mapdepth == 24:
  109.         self.palette = ImagePalette.raw("BGR",
  110.             "\0"*3*start + self.fp.read(3*size))
  111.         elif mapdepth == 32:
  112.         self.palette = ImagePalette.raw("BGRA",
  113.             "\0"*4*start + self.fp.read(4*size))
  114.  
  115.     # setup tile descriptor
  116.     try:
  117.         rawmode = MODES[(imagetype&7, depth)]
  118.         if imagetype & 8:
  119.         # compressed
  120.         self.tile = [("tga_rle", (0, 0)+self.size,
  121.                   self.fp.tell(), (rawmode, orientation, depth))]
  122.         else:
  123.         self.tile = [("raw", (0, 0)+self.size,
  124.                   self.fp.tell(), (rawmode, 0, orientation))]
  125.     except KeyError:
  126.         pass # cannot decode
  127.  
  128. #
  129. # registry
  130.  
  131. Image.register_open("TGA", TgaImageFile, _accept)
  132.  
  133. Image.register_extension("TGA", ".tga")
  134.